from monai.networks.nets import UNETR, SwinUNETR, UNet, VNet, SegResNet, AttentionUnet